Post-Training is About States, Not Tokens: A State Distribution View of SFT, RL, and On-Policy Distillation
后训练关键在状态分布而非Token:SFT、RL与在线蒸馏的状态视角
🧭 LLM后训练状态分布视角 ★ 深度解读
提出状态分布视角统一SFT、RL和在线蒸馏,实验显示学生可超越退化教师且on-policy方法更稳定
📌 工作参考价值(4 点)
- 用状态源/信号源拆解训练策略:将推荐Agent的训练明确分为“状态采自哪里”和“监督信号从哪里来”。离线数据(如用户历史日志)提供固定状态,但Agent部署后可能遇到未覆盖的状态分布,借鉴on-policy采样(RL或在线蒸馏)能减轻分布不匹配造成的性能退化。
- 在线蒸馏时让学生控制状态分布:在电商场景中,Teacher模型可能在某些上下文(如长对话或冷门商品)表现差,但局部Token预测仍可用。让学生Agent在自身状态上请求教师指导(如生成短续写),类似DAgger,可以保留教师局部能力的同时避免复制教师的全局错误轨迹,实现学生超越教师。
- 避免单纯用标量漂移判断遗忘:业务监控时,不要仅靠rollout状态的MMD等单一距离指标下结论。论文显示,两个模型可能有相近的全局状态漂移但遗忘程度迥异。应结合状态来源分析——是在模型自身可达状态上做更新(如RL/OPD),还是在与模型当前策略偏离的固定状态上做密集离线更新(如stress SFT)。
- 轻量在线RL可作为能力保留手段:在追求目标指标提升时,可以先跑一轮温和SFT注入基础能力,再用在线RL(如带分组相对奖励的GRPO)在模型自身轨迹上小步优化,既能提升主任务,又可抑制遗忘。对电商搜索排序或对话Agent,可设定简单的验证性奖励(如用户点击、转化),用on-policy采样来微调,保持通用知识不退化。